#ifndef PHOTON_MAP_H
#define PHOTON_MAP_H

#include "src/scene/object.h"
#include "src/scene/kd/kdtree.h"
#include "src/scene/photon.h"

class PhotonMap{
    public:
        PhotonMap(Object *);
        ~PhotonMap();

        Object *getObject() const {return object;};
        KdTree *getKdTree() const {return kdTree;};

        void addPhoton(Photon *_p) {kdTree->addPhoton(_p);};
        void createHeap() {kdTree->createHeap();};

    private:
        Object *object;
        KdTree *kdTree;
};

#endif

